package com.spring.controller;

import org.springframework.web.bind.annotation.PostMapping;
import org.springframework.web.bind.annotation.RestController;
import org.springframework.web.multipart.MultipartFile;

import javax.servlet.http.HttpServletRequest;
import java.io.File;
import java.io.IOException;
import java.text.SimpleDateFormat;
import java.util.ArrayList;
import java.util.Date;
import java.util.List;
import java.util.UUID;

@RestController
public class FlieUploadController {
    SimpleDateFormat sdf=new SimpleDateFormat("yyyy/MM/dd/");
    @PostMapping("/upload")
    public String  upload(MultipartFile uploadFile, HttpServletRequest request){
        String realPath=request.getSession().getServletContext().getRealPath("/uploadFile/");
        String format=sdf.format(new Date());
        File folder=new File(realPath+format);
        if (!folder.isDirectory()){
            folder.mkdirs();
        }
        String oldName=uploadFile.getOriginalFilename();
        String newName= UUID.randomUUID().toString()+oldName.substring(oldName.lastIndexOf("."),oldName.length());
        try{
            uploadFile.transferTo(new File(folder,newName));
            String filePath=request.getScheme()+"://"+request.getServerName()+":"+request.getServerPort()+"/uploadFile/"+format+newName;
            return filePath;
        }catch (IOException e){
            e.printStackTrace();
        }
        return "上传失败";
    }
    @PostMapping("/uploads")
    public  List<String>  uploads(MultipartFile[] uploadFiles, HttpServletRequest request){
        String realPath=request.getSession().getServletContext().getRealPath("/uploadFile/");
        String format=sdf.format(new Date());
        File folder=new File(realPath+format);
        if (!folder.isDirectory()){
            folder.mkdirs();
        }
        List<String> filePath=new ArrayList<>();
        for (MultipartFile upload:uploadFiles) {
            String oldName=upload.getOriginalFilename();
            String newName= UUID.randomUUID().toString()+oldName.substring(oldName.lastIndexOf("."),oldName.length());
            try{
                upload.transferTo(new File(folder,newName));
                filePath.add(request.getScheme()+"://"+request.getServerName()+":"+request.getServerPort()+"/uploadFile/"+format+newName);
            }catch (IOException e){
                e.printStackTrace();
            }
        }
        return filePath;
    }
}
